package user;

import java.io.IOException;
import java.io.PrintWriter;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

public class Login extends HttpServlet {

	/**
	 * Constructor of the object.
	 */
	public Login() {
		super();
	}

	/**
	 * Destruction of the servlet. <br>
	 */
	public void destroy() {
		super.destroy(); // Just puts "destroy" string in log
		// Put your code here
	}

	/**
	 * The doGet method of the servlet. <br>
	 *
	 * This method is called when a form has its tag value method equals to get.
	 * 
	 * @param request the request send by the client to the server
	 * @param response the response send by the server to the client
	 * @throws ServletException if an error occurred
	 * @throws IOException if an error occurred
	 */
	public void doGet(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {

	}

	/**
	 * The doPost method of the servlet. <br>
	 *
	 * This method is called when a form has its tag value method equals to post.
	 * 
	 * @param request the request send by the client to the server
	 * @param response the response send by the server to the client
	 * @throws ServletException if an error occurred
	 * @throws IOException if an error occurred
	 */
	public void doPost(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {
		response.setContentType("text/html");
		PrintWriter out = response.getWriter();
		//out.println("<!DOCTYPE HTML PUBLIC \"-//W3C//DTD HTML 4.01 Transitional//EN\">");
		//out.println("<HTML>");
		//out.println("  <HEAD><TITLE>A Servlet</TITLE></HEAD>");
		//out.println("  <BODY>");
		
		String type = request.getParameter("user_type");
		String username = request.getParameter("username");
		String password = request.getParameter("password");
		
		if (type.equals("1")) {
			User user = new Customer();
			user.setUsername(username);
			user.setPassword(password);
			LoginValidation validation = new LoginValidation(user, type);
			boolean isValid = validation.isValidate();
			
			if (isValid) {
				HttpSession session = request.getSession(true);
				session.setAttribute("uType", type);
				session.setAttribute("id", LoginValidation.getUserId(username));
				session.setAttribute("username", username);
				response.sendRedirect("../index.jsp");
			} else {
				/*
				 * out.println("<script language='javascript'>" +
						"alert('Username doesn't exist or wrong password! Please register first or check you password!');" +
						"window.location.href = '../register.html';" + 
						"</script>");
				 */
				
				response.sendRedirect("../register.html");
			}		
		} else if (type.equals("2")) {
			User user = new RestaurantOwner();
			user.setUsername(username);
			user.setPassword(password);
			LoginValidation validation = new LoginValidation(user, type);
			boolean isValid = validation.isValidate();
			
			if (isValid) {
				HttpSession session = request.getSession(true);
				session.setAttribute("uType", type);
				// TODO change it to rest_id
				session.setAttribute("id", LoginValidation.getRestaurantId(username));
				session.setAttribute("username", username);
				response.sendRedirect("../restaurant_account.jsp");
			} else {
				/*
				 * out.println("<script language='javascript'>" +
						"alert('Username doesn't exist or wrong password! Please register first or check you password!');" +
						"window.location.href = '../register.html';" + 
						"</script>");
				 */
				
				response.sendRedirect("../register.html");
			}		
			
		} else {
			System.out.println("admin");
			if (username.equals("Lumen") && password.equals("lumen")) {
				System.out.println("inside if");
				HttpSession session = request.getSession(true);
				session.setAttribute("uType", type);
				session.setAttribute("id", "1");
				session.setAttribute("username", username);
				response.sendRedirect("../administrator_account.jsp");
			} else {
				/*
				 * out.println("<script language='javascript'>" +
						"alert('Username doesn't exist or wrong password! Please register first or check you password!');" +
						"window.location.href = '../register.html';" + 
						"</script>");
				 */
				
				response.sendRedirect("../register.html");
			}
			
		}
		
		
		//out.println("  </BODY>");
		//out.println("</HTML>");
		//out.flush();
		//out.close();
	}

	/**
	 * Initialization of the servlet. <br>
	 *
	 * @throws ServletException if an error occurs
	 */
	public void init() throws ServletException {
		// Put your code here
	}

}
